package com.zwq.leetcode.arr.twonumsum;

import java.util.Arrays;
import java.util.HashMap;

/**
 * 给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的那 两个 整数，并返回他们的数组下标。
 *
 * 你可以假设每种输入只会对应一个答案。但是，数组中同一个元素不能使用两遍。
 *
 * 示例:
 *
 * 给定 nums = [2, 7, 11, 15], target = 9
 *
 * 因为 nums[0] + nums[1] = 2 + 7 = 9
 * 所以返回 [0, 1]
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/two-sum
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
public class Solution {
    public static void main(String[] args) {
        int [] nums = {2,7,11,15};
        int target = 9;
        int[] result = solution2(nums, target);
        System.out.println(result[0] + "-" +result[1]);
    }

    private static int [] solution1(int[] nums, int target){
        for(int i =0 ; i < nums.length ; i++){
            for (int j = i + 1;j<nums.length;j++){
                if(nums[j] == target - nums[i]){
                    return new int[]{i,j};
                }
            }
        }
        return null;
    }

    private static int [] solution2(int[] nums, int target){
        HashMap<Integer,Integer> map = new HashMap();
        int []res = new int[2];
        if (nums == null && nums.length <= 1) return res;
        for(int i=0;i<nums.length;i++){
            if(map.containsKey(target - nums[i])){
                res[0] = map.get(target - nums[i]);
                res[1] = i;
                // return new int[]{map.get(target - nums[i]),i};
                return res;
            }
            map.put(nums[i],i);
        }
        return res;
    }

    /**
     * 有序的数组，如果数组无序，需要先排序并记录下标映射
     * @param nums
     * @param target
     * @return
     */
    private static int[] solution(int[] nums, int target) {

        //排序，记录下标映射
        HashMap<Integer,Integer> indexMap = new HashMap<>();
        for(int i = 0;i <  nums.length ;i++){
            indexMap.put(nums[i],i);
        }
        Arrays.sort(nums);

        int begin = 0;
        int end = nums.length - 1;
        while(begin < end){
            if(nums[begin] + nums[end] > target){
                end --;
            }else if(nums[begin] + nums[end] < target){
                begin ++;
            }else{
                break;
            }
        }
        return begin  == end ? null :  new int[]{indexMap.get(nums[begin]),indexMap.get(nums[end]) } ;
    }
}
